package com.pai4j.sso.handler;

import jakarta.annotation.Resource;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 *   微信消息处理工厂
 *
 * @author: CYM-pai
 * @date: 2025/04/16 17:07
 **/
@Component
public class WxChatMsgFactory implements InitializingBean {
   
    // 使用Resource注解注入WxChatMsgHandler的列表，用于处理不同的微信消息类型
    @Resource
    private List<WxChatMsgHandler> wxChatMsgHandlerList;
    
    // 创建一个映射，用于根据消息类型快速获取对应的处理handler
    private Map<WxChatMsgTypeEnum, WxChatMsgHandler> handlerMap = new HashMap<>();
    
    /**
     * 根据消息类型获取处理该消息的handler
     *
     * @param msgType 消息类型字符串，用于识别具体的微信聊天消息类型
     * @return 返回对应的WxChatMsgHandler实例，如果没有找到则返回null
     */
    public WxChatMsgHandler getHandlerByMsgType(String msgType) {
        // 根据消息类型字符串获取对应的枚举类型
        WxChatMsgTypeEnum msgTypeEnum = WxChatMsgTypeEnum.getByMsgType(msgType);
        // 根据枚举类型从映射中获取并返回对应的handler
        return handlerMap.get(msgTypeEnum);
    }
    
    /**
     * 在所有必要属性设置完成后执行此方法，用于初始化handler映射
     *
     * @throws Exception 属性设置过程中可能抛出的异常
     */
    @Override
    public void afterPropertiesSet() throws Exception {
        // 遍历注入的微信消息处理handler列表
        for (WxChatMsgHandler wxChatMsgHandler : wxChatMsgHandlerList) {
            // 将消息类型枚举和对应的handler实例添加到映射中，以便快速查找
            handlerMap.put(wxChatMsgHandler.getMsgType(), wxChatMsgHandler);
        }
    }

}
